package tasks.leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * • @：表示换行
 * • ~：表示删除模式，如果光标没有显示句末，删除当前光标所在的元素，后面的元素往前移动
 * • ^：光标上移动
 * • <：向左移动光标一位
 * • >：向右移动光标一位
 *
 * 使用java，便携一个这样的文本编辑器的程序，用户输入一个字符串例如:aaaa@bb<<<^bbbb，最终输出一个目标的字符串数组
 * 代码是 通义千问实现的，太强了！
 */
public class SimpleTextEditor {
    public static String[] processCommands(String commands) {
        List<StringBuilder> lines = new ArrayList<>();
        lines.add(new StringBuilder());

        int cursorX = 0; // 光标在当前行的位置
        int cursorY = 0; // 光标所在的行号

        for (int i = 0; i < commands.length(); i++) {
            char command = commands.charAt(i);

            if (command == '@') {
                // 换行
                if (cursorY + 1 >= lines.size()) {
                    lines.add(new StringBuilder());
                }
                cursorY++;
                cursorX = 0;
            } else if (command == '~') {
                // 删除模式
                if (cursorX < lines.get(cursorY).length()) {
                    lines.get(cursorY).deleteCharAt(cursorX);
                }
            } else if (command == '^') {
                // 光标上移一行
                if (cursorY > 0) {
                    cursorY--;
                    cursorX = Math.min(cursorX, lines.get(cursorY).length());
                }
            } else if (command == '<') {
                // 向左移动光标一位
                if (cursorX > 0) {
                    cursorX--;
                }
            } else if (command == '>') {
                // 向右移动光标一位
                if (cursorX < lines.get(cursorY).length()) {
                    cursorX++;
                }
            } else {
                // 插入普通字符
                lines.get(cursorY).insert(cursorX, command);
                cursorX++;
            }
        }

        // 转换成字符串数组
        String[] result = new String[lines.size()];
        for (int i = 0; i < lines.size(); i++) {
            result[i] = lines.get(i).toString();
        }
        return result;
    }

    public static void main(String[] args) {
        String commands = "aaaa@bb<<<^~~bbb";
        String[] result = processCommands(commands);
        for (String line : result) {
            System.out.println(line);
        }
    }
}